import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
import os
from tensorflow.keras import layers


train_data = np.loadtxt('data/train.csv', delimiter=',', skiprows=1)
test_data = np.loadtxt('data/test.csv', delimiter=',', skiprows=1)

print(train_data.shape)